
import wandb

sweep_configuration = {
    'method': 'random',
    'metric': {'goal': 'minimize', 'name': 'loss'},
    'parameters':
    {
        'x0': {'values': [[0.1,0.1]]},
        'lr': {'values': [0.2]},
        'optimizer': {'values': ['Adam','Adagrad','RMSProp']},
        'epochs':{'values': [15]},
     }
}

